#pragma once
#include <string>
#include <iostream>
#include <vector>
#include <atomic>
#include <fstream>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <sys/time.h>
// #include <boost/algorithm/string.hpp>

namespace ns_util
{
    // 临时文件存放路径
    // 注意：该路径必须存在，且有读写权限
    const std::string temp_path = "./temp/";

    // 时间功能
    class TimeUtil
    {
    public:
        // 获取秒级别时间戳
        static std::string GetTimeStamp()
        {
            struct timeval _time;
            gettimeofday(&_time, nullptr); // 时区不关心，所以设置成nullptr

            return std::to_string(_time.tv_sec);
        }

        // 获取毫秒级别时间戳
        static std::string GetTimeMs()
        {
            struct timeval _time;
            gettimeofday(&_time, nullptr);
            return std::to_string(_time.tv_sec * 1000 + _time.tv_usec / 1000); // 秒->毫秒=秒*1000，微秒->毫秒=微秒/1000
        }
    };

    // 对路径操作的方法
    class PathUtil
    {
    public:
        // 构建文件名+后缀的完整文件名

        static std::string AddSuffix(const std::string &file_name, const std::string &suffix)
        {
            std::string path_name = temp_path;
            path_name += file_name;
            path_name += suffix;
            return path_name;
        }

        /**************************编译时需要有的临时文件**************************/

        // 构建源文件路径+后缀的完整文件名
        // 比如： 1234 ->./temp/1234.cpp
        static std::string Src(const std::string &file_name)
        {
            return AddSuffix(file_name, ".cpp");
        }
        // 构建可执行文件路径+后缀的完整文件名
        // 比如： 1234 ->./temp/1234.exe
        static std::string Exe(const std::string &file_name)
        {
            return AddSuffix(file_name, ".exe");
        }

        static std::string CompilerError(const std::string &file_name)
        {
            return AddSuffix(file_name, ".compiler_error");
        }

        /**************************运行时需要有的临时文件**************************/

        // 构建该程序对应的标准输出文件路径+后缀的完整文件名
        // 比如： 1234 ->./temp/1234.out
        static std::string Stderr(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stderr");
        }
        // 构建该程序对应的标准输出文件路径+后缀的完整文件名
        // 比如： 1234 ->./temp/1234.stdout
        static std::string Stdout(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdout");
        }
        // 构建该程序对应的标准输入文件路径+后缀的完整文件名
        // 比如： 1234 ->./temp/1234.stdin
        static std::string Stdin(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdin");
        }
    };

    // 对文件操作的方法
    class FileUtil
    {
    public:
        // 判断文件是否存在
        static bool IsFileExists(const std::string &path_name)
        {
            struct stat st;
            if (stat(path_name.c_str(), &st) == 0)
            {
                // 获取属性成功，表示文件已经存在
                return true;
            }

            return false;
        }

        // 形成一个唯一的文件名，没有目录没有后缀
        // 毫秒级时间戳+原子性递增唯一值: 来保证唯一性
        static std::string UniqFileName()
        {
            static std::atomic_uint id(0); // 原子性的计数器
            id++;                          // 计数器++

            std::string ms = TimeUtil::GetTimeMs();   // 毫秒级时间戳
            std::string uniq_id = std::to_string(id); // 获得唯一id

            return ms + "_" + uniq_id;
        }

        // 将代码写入该文件
        // 参数：target为要写入的文件名，content为要写入的内容
        static bool WriteFile(const std::string &target, std::string &content)
        {
            std::ofstream out(target);
            if (!out.is_open())
            {
                // 如果没有被打开成功
                return false;
            }

            out.write(content.c_str(), content.size()); // 写入内容
            out.close();                                // 关闭文件

            return true;
        }

        // 把所有的文件内容读出来
        // 参数：target为要读取的文件名，content用于存储读到的数据, keep用于判断是否保留\n
        static bool ReadFile(const std::string &target, std::string *content, bool keep = false)
        {
            (*content).clear(); // 先清空content，为了不影响第一次读取

            std::ifstream in(target, std::ios::binary);
            if (!in.is_open())
            {
                return false;
            }

            std::string line;
            // 注意：
            // getline:不能保存行分隔符
            // getline:内部重载了强制类型转化
            while (std::getline(in, line))
            {
                (*content) += line;
                (*content) += (keep ? "\n" : ""); // 判断是否保留\n
            }

            in.close(); // 关闭文件

            return true;
        }
    };
    // 字符串工具
    class StringUtil
    {
    public:
        // 切分字符串
        /******************************************************
         * 参数：
         * str：输入型，目标要切分的字符串
         * target：输出型，保存切分完毕的结果
         * sep：指定的分隔符
         ******************************************************/
        static void SplitString(const std::string &str, std::vector<std::string> *target, const std::string sep)
        {
            // 用boost库里面的字符串切分功能
            boost::split((*target), str, boost::is_any_of(sep), boost::algorithm::token_compress_on);
        }
    };
}